Linear search in python